#!/usr/bin/env bash

# set some environment variables
ORG_PATH="go.etcd.io"
REPO_PATH="${ORG_PATH}/etcd/v3"

GIT_SHA=$(git rev-parse --short HEAD || echo "GitNotFound")
if [[ -n "$FAILPOINTS" ]]; then
	GIT_SHA="$GIT_SHA"-FAILPOINTS
fi

# Set GO_LDFLAGS="-s" for building without symbols for debugging.
GO_LDFLAGS="$GO_LDFLAGS -X ${REPO_PATH}/version.GitSHA=${GIT_SHA}"

# enable/disable failpoints
toggle_failpoints() {
	mode="$1"
	if command -v gofail >/dev/null 2>&1; then
		gofail "$mode" etcdserver/ mvcc/backend/
	elif [[ "$mode" != "disable" ]]; then
		echo "FAILPOINTS set but gofail not found"
		exit 1
	fi
}

toggle_failpoints_default() {
	mode="disable"
	if [[ -n "$FAILPOINTS" ]]; then mode="enable"; fi
	toggle_failpoints "$mode"
}

etcd_build() {
	out="bin"
	if [[ -n "${BINDIR}" ]]; then out="${BINDIR}"; fi
	toggle_failpoints_default

	# Static compilation is useful when etcd is run in a container. $GO_BUILD_FLAGS is OK
	# shellcheck disable=SC2086
	CGO_ENABLED=0 go build $GO_BUILD_FLAGS \
		-installsuffix cgo \
		-ldflags "$GO_LDFLAGS" \
		-o "${out}/etcd" ${REPO_PATH} || return
	# shellcheck disable=SC2086
	CGO_ENABLED=0 go build $GO_BUILD_FLAGS \
		-installsuffix cgo \
		-ldflags "$GO_LDFLAGS" \
		-o "${out}/etcdctl" ${REPO_PATH}/etcdctl || return
}

tools_build() {
	out="bin"
	if [[ -n "${BINDIR}" ]]; then out="${BINDIR}"; fi
	tools_path="tools/benchmark
	tools/etcd-dump-db
	tools/etcd-dump-logs
	tools/local-tester/bridge
	functional/cmd/etcd-agent
	functional/cmd/etcd-proxy
	functional/cmd/etcd-runner
	functional/cmd/etcd-tester"
	for tool in ${tools_path}
	do
		echo "Building" "'${tool}'"...
		# shellcheck disable=SC2086
		CGO_ENABLED=0 go build ${GO_BUILD_FLAGS} \
		  -installsuffix cgo \
		  -ldflags "${GO_LDFLAGS}" \
		  -o "${out}/${tool}" "${REPO_PATH}/${tool}" || return
	done
}

toggle_failpoints_default

# only build when called directly, not sourced
if echo "$0" | grep "build$" >/dev/null; then
	etcd_build
fi
